calculate transpose of an matrix in python